from SkipGram import SkipGram
import torch
net = torch.load("../dist/skip-gram.pkl")
print("\"share\"与\"enjoyment\"的相似度为:{:.3f}".format(net.predict("share", "enjoyment")))